package array;

/**
 * @author tong
 * @date 2021/12/21 10:40
 * 稀疏数组
 */
public class SparseArray {
    public static void main(String[] args) {
        // 初始化数组
        int[][] initArray = new int[11][11];
        initArray[1][2] = 1;
        initArray[2][3] = 2;
        printArray(initArray);
        // 二维数组->稀疏数组
        int[][] sparseArray = arrayToSparseArray(initArray);
        printArray(sparseArray);
        // 稀疏数组->二维数组
        int[][] array = sparseArrayToArray(sparseArray);
        printArray(array);
    }

    private static int[][] arrayToSparseArray(int[][] initArray) {
        int sum = 0;
        for (int[] row : initArray) {
            for (int data : row) {
                if (data != 0) {
                    sum++;
                }
            }
        }
        // 先确定数组大小
        int[][] sparseArray = new int[sum + 1][3];
        sparseArray[0][0] = 11;
        sparseArray[0][1] = 11;
        sparseArray[0][2] = sum;
        // 从第二行填充
        int rowCount = 1;
        for (int i = 0; i < initArray.length; i++) {
            for (int j = 0; j < initArray[i].length; j++) {
                if (initArray[i][j] != 0) {
                    sparseArray[rowCount][0] = i;
                    sparseArray[rowCount][1] = j;
                    sparseArray[rowCount][2] = initArray[i][j];
                    rowCount++;
                }
            }
        }
        return sparseArray;
    }

    private static int[][] sparseArrayToArray(int[][] sparseArray) {
        int rowNum = sparseArray[0][0];
        int cowNum = sparseArray[0][1];
        int valueNum = sparseArray[0][2];
        int[][] array = new int[rowNum][cowNum];
        for (int i = 1; i <= valueNum; i++) {
            int row = sparseArray[i][0];
            int cow = sparseArray[i][1];
            int value = sparseArray[i][2];
            array[row][cow] = value;
        }
        return array;
    }

    private static void printArray(int[][] array) {
        for (int[] row : array) {
            for (int data : row) {
                System.out.printf("%d\t", data);
            }
            System.out.println();
        }
    }
}
